package com.kk.flyweight;

import java.util.HashMap;

/**
 * @author 王志强
 * @date 2022-01-06
 */
public class WebSiteFactory {

    /**
     * 存放各种网站
     */
    private final HashMap<String, ConcreteWebSite> pool = new HashMap<>();


    /**
     * 获得不同网站的个数
     *
     * @return
     */
    public int poolSize() {
        return pool.size();
    }

    /**
     * 返回具体的网站
     *
     * @param type
     * @return
     */
    public WebSite getWebSiteCategory(String type) {
        if (!pool.containsKey(type)) {
            pool.put(type, new ConcreteWebSite(type));
        }
        return pool.get(type);
    }

}
